import torch, torchvision
from torch import nn
from torch.nn import L1Loss
import torch.nn.functional as F

# 模型的保存

# vgg16 = torchvision.models.vgg16(pretrained=True)

# torch.save(vgg16, 'vgg16_model.pth')

vgg16_load = torch.load('vgg16_model.pth')

print(vgg16_load)